# pip install ragas
from ragas import evaluate
import os
from datasets import Dataset
from ragas.metrics import context_precision

os.environ["OPENAI_API_KEY"] = "your api key"
data = {
    "question": ["法国的首都在哪里?"],  # 问题
    "answer": ["巴黎"],  # RAG给出的回答
    "contexts": [["法国是一个在西欧的国家", "法国首都是巴黎"]],  # 检索到的上下文
    "ground_truth": ["法国的首都是巴黎"],  # 真实答案
}
# 把数据字典转换为HuggingFace的Dataset对象
dataset = Dataset.from_dict(data)

result = evaluate(dataset, metrics=[context_precision])
print(result)
